home *** CD-ROM | disk | FTP | other *** search
/ Mac-Source 1994 July / Mac-Source_July_1994.iso / Other Langs / MacPerl ƒ / Perl Source ƒ / Perl / cons.c < prev    next >
Text File  |  1993-10-23  |  37KB  |  1,465 lines
  1. /* $RCSfile: cons.c,v $$Revision: 4.0.1.4 $$Date: 1993/02/05 19:30:15 $
  2.  *
  3.  *    Copyright (c) 1991, Larry Wall
  4.  *
  5.  *    You may distribute under the terms of the Perl Artistic License,
  6.  *    as specified in the README file.
  7.  *
  8.  * $Log:    cons.c,v $
  9.  * Revision 4.0.1.3  1992/06/08  12:18:35  lwall
  10.  * patch20: removed implicit int declarations on funcions
  11.  * patch20: deleted some minor memory leaks
  12.  * patch20: fixed double debug break in foreach with implicit array assignment
  13.  * patch20: fixed confusion between a *var's real name and its effective name
  14.  * patch20: Perl now distinguishes overlapped copies from non-overlapped
  15.  * patch20: debugger sometimes displayed wrong source line
  16.  * patch20: various error messages have been clarified
  17.  * patch20: an eval block containing a null block or statement could dump core
  18.  * 
  19.  * Revision 4.0.1.2  91/11/05  16:15:13  lwall
  20.  * patch11: debugger got confused over nested subroutine definitions
  21.  * patch11: prepared for ctype implementations that don't define isascii()
  22.  * 
  23.  * Revision 4.0.1.1  91/06/07  10:31:15  lwall
  24.  * patch4: new copyright notice
  25.  * patch4: added global modifier for pattern matches
  26.  * 
  27.  * Revision 4.0  91/03/20  01:05:51  lwall
  28.  * 4.0 baseline.
  29.  * 
  30.  */
  31.  
  32. #include "EXTERN.h"
  33. #include "perl.h"
  34. #include "perly.h"
  35.  
  36. extern char *tokename[];
  37. extern int yychar;
  38.  
  39. static int cmd_tosave();
  40. static int arg_tosave();
  41. static int spat_tosave();
  42. static void make_cswitch();
  43. static void make_nswitch();
  44.  
  45. static bool saw_return;
  46.  
  47. SUBR *
  48. make_sub(name,cmd)
  49. char *name;
  50. CMD *cmd;
  51. {
  52.     register SUBR *sub;
  53.     STAB *stab = stabent(name,TRUE);
  54.  
  55.     if (sub = stab_sub(stab)) {
  56.     if (dowarn) {
  57.         CMD *oldcurcmd = curcmd;
  58.  
  59.         if (cmd)
  60.         curcmd = cmd;
  61.         warn("Subroutine %s redefined",name);
  62.         curcmd = oldcurcmd;
  63.     }
  64.     if (!sub->usersub && sub->cmd) {
  65.         cmd_free(sub->cmd);
  66.         sub->cmd = Nullcmd;
  67.         afree(sub->tosave);
  68.     }
  69.     Safefree(sub);
  70.     }
  71.     Newz(101,sub,1,SUBR);
  72.     stab_sub(stab) = sub;
  73.     sub->filestab = curcmd->c_filestab;
  74.     saw_return = FALSE;
  75.     tosave = anew(Nullstab);
  76.     tosave->ary_fill = 0;    /* make 1 based */
  77.     (void)cmd_tosave(cmd,FALSE);    /* this builds the tosave array */
  78.     sub->tosave = tosave;
  79.     if (saw_return) {
  80.     struct compcmd mycompblock;
  81.  
  82.     mycompblock.comp_true = cmd;
  83.     mycompblock.comp_alt = Nullcmd;
  84.     cmd = add_label(savestr("_SUB_"),make_ccmd(C_BLOCK,0,
  85.         Nullarg,mycompblock));
  86.     saw_return = FALSE;
  87.     cmd->c_flags |= CF_TERM;
  88.     cmd->c_head = cmd;
  89.     }
  90.     sub->cmd = cmd;
  91.     if (perldb) {
  92.     STR *str;
  93.     STR *tmpstr = str_mortal(&str_undef);
  94.  
  95.     sprintf(buf,"%s:%ld",stab_val(curcmd->c_filestab)->str_ptr, subline);
  96.     str = str_make(buf,0);
  97.     str_cat(str,"-");
  98.     sprintf(buf,"%ld",(long)curcmd->c_line);
  99.     str_cat(str,buf);
  100.     stab_efullname(tmpstr,stab);
  101.     hstore(stab_xhash(DBsub), tmpstr->str_ptr, tmpstr->str_cur, str, 0);
  102.     }
  103.     Safefree(name);
  104.     return sub;
  105. }
  106.  
  107. SUBR *
  108. make_usub(name, ix, subaddr, filename)
  109. char *name;
  110. int ix;
  111. int (*subaddr)();
  112. char *filename;
  113. {
  114.     register SUBR *sub;
  115.     STAB *stab = stabent(name,allstabs);
  116.  
  117.     if (!stab)                /* unused function */
  118.     return Null(SUBR*);
  119.     if (sub = stab_sub(stab)) {
  120.     if (dowarn)
  121.         warn("Subroutine %s redefined",name);
  122.     if (!sub->usersub && sub->cmd) {
  123.         cmd_free(sub->cmd);
  124.         sub->cmd = Nullcmd;
  125.         afree(sub->tosave);
  126.     }
  127.     Safefree(sub);
  128.     }
  129.     Newz(101,sub,1,SUBR);
  130.     stab_sub(stab) = sub;
  131.     sub->filestab = fstab(filename);
  132.     sub->usersub = subaddr;
  133.     sub->userindex = ix;
  134.     return sub;
  135. }
  136.  
  137. void
  138. make_form(stab,fcmd)
  139. STAB *stab;
  140. FCMD *fcmd;
  141. {
  142.     if (stab_form(stab)) {
  143.     FCMD *tmpfcmd;
  144.     FCMD *nextfcmd;
  145.  
  146.     for (tmpfcmd = stab_form(stab); tmpfcmd; tmpfcmd = nextfcmd) {
  147.         nextfcmd = tmpfcmd->f_next;
  148.         if (tmpfcmd->f_expr)
  149.         arg_free(tmpfcmd->f_expr);
  150.         if (tmpfcmd->f_unparsed)
  151.         str_free(tmpfcmd->f_unparsed);
  152.         if (tmpfcmd->f_pre)
  153.         Safefree(tmpfcmd->f_pre);
  154.         Safefree(tmpfcmd);
  155.     }
  156.     }
  157.     stab_form(stab) = fcmd;
  158. }
  159.  
  160. CMD *
  161. block_head(tail)
  162. register CMD *tail;
  163. {
  164.     CMD *head;
  165.     register int opt;
  166.     register int last_opt = 0;
  167.     register STAB *last_stab = Nullstab;
  168.     register int count = 0;
  169.     register CMD *switchbeg = Nullcmd;
  170.  
  171.     if (tail == Nullcmd) {
  172.     return tail;
  173.     }
  174.     head = tail->c_head;
  175.  
  176.     for (tail = head; tail; tail = tail->c_next) {
  177.  
  178.     /* save one measly dereference at runtime */
  179.     if (tail->c_type == C_IF) {
  180.         if (!(tail->ucmd.ccmd.cc_alt = tail->ucmd.ccmd.cc_alt->c_next))
  181.         tail->c_flags |= CF_TERM;
  182.     }
  183.     else if (tail->c_type == C_EXPR) {
  184.         ARG *arg;
  185.  
  186.         if (tail->ucmd.acmd.ac_expr)
  187.         arg = tail->ucmd.acmd.ac_expr;
  188.         else
  189.         arg = tail->c_expr;
  190.         if (arg) {
  191.         if (arg->arg_type == O_RETURN)
  192.             tail->c_flags |= CF_TERM;
  193.         else if (arg->arg_type == O_ITEM && arg[1].arg_type == A_CMD)
  194.             tail->c_flags |= CF_TERM;
  195.         }
  196.     }
  197.     if (!tail->c_next)
  198.         tail->c_flags |= CF_TERM;
  199.  
  200.     if (tail->c_expr && (tail->c_flags & CF_OPTIMIZE) == CFT_FALSE)
  201.         opt_arg(tail,1, tail->c_type == C_EXPR);
  202.  
  203.     /* now do a little optimization on case-ish structures */
  204.     switch(tail->c_flags & (CF_OPTIMIZE|CF_FIRSTNEG|CF_INVERT)) {
  205.     case CFT_ANCHOR:
  206.     case CFT_STROP:
  207.         opt = (tail->c_flags & CF_NESURE) ? CFT_STROP : 0;
  208.         break;
  209.     case CFT_CCLASS:
  210.         opt = CFT_STROP;
  211.         break;
  212.     case CFT_NUMOP:
  213.         opt = (tail->c_slen == O_NE ? 0 : CFT_NUMOP);
  214.         if ((tail->c_flags&(CF_NESURE|CF_EQSURE)) != (CF_NESURE|CF_EQSURE))
  215.         opt = 0;
  216.         break;
  217.     default:
  218.         opt = 0;
  219.     }
  220.     if (opt && opt == last_opt && tail->c_stab == last_stab)
  221.         count++;
  222.     else {
  223.         if (count >= 3) {        /* is this the breakeven point? */
  224.         if (last_opt == CFT_NUMOP)
  225.             make_nswitch(switchbeg,count);
  226.         else
  227.             make_cswitch(switchbeg,count);
  228.         }
  229.         if (opt) {
  230.         count = 1;
  231.         switchbeg = tail;
  232.         }
  233.         else
  234.         count = 0;
  235.     }
  236.     last_opt = opt;
  237.     last_stab = tail->c_stab;
  238.     }
  239.     if (count >= 3) {        /* is this the breakeven point? */
  240.     if (last_opt == CFT_NUMOP)
  241.         make_nswitch(switchbeg,count);
  242.     else
  243.         make_cswitch(switchbeg,count);
  244.     }
  245.     return head;
  246. }
  247.  
  248. /* We've spotted a sequence of CMDs that all test the value of the same
  249.  * spat.  Thus we can insert a SWITCH in front and jump directly
  250.  * to the correct one.
  251.  */
  252. static void
  253. make_cswitch(head,count)
  254. register CMD *head;
  255. int count;
  256. {
  257.     register CMD *cur;
  258.     register CMD **loc;
  259.     register int i;
  260.     register int min = 255;
  261.     register int max = 0;
  262.  
  263.     /* make a new head in the exact same spot */
  264.     New(102,cur, 1, CMD);
  265.     StructCopy(head,cur,CMD);
  266.     Zero(head,1,CMD);
  267.     head->c_head = cur->c_head;
  268.     head->c_type = C_CSWITCH;
  269.     head->c_next = cur;        /* insert new cmd at front of list */
  270.     head->c_stab = cur->c_stab;
  271.  
  272.     Newz(103,loc,258,CMD*);
  273.     loc++;                /* lie a little */
  274.     while (count--) {
  275.     if ((cur->c_flags & CF_OPTIMIZE) == CFT_CCLASS) {
  276.         for (i = 0; i <= 255; i++) {
  277.         if (!loc[i] && cur->c_short->str_ptr[i>>3] & (1 << (i&7))) {
  278.             loc[i] = cur;
  279.             if (i < min)
  280.             min = i;
  281.             if (i > max)
  282.             max = i;
  283.         }
  284.         }
  285.     }
  286.     else {
  287.         i = *cur->c_short->str_ptr & 255;
  288.         if (!loc[i]) {
  289.         loc[i] = cur;
  290.         if (i < min)
  291.             min = i;
  292.         if (i > max)
  293.             max = i;
  294.         }
  295.     }
  296.     cur = cur->c_next;
  297.     }
  298.     max++;
  299.     if (min > 0)
  300.     Move(&loc[min],&loc[0], max - min, CMD*);
  301.     loc--;
  302.     min--;
  303.     max -= min;
  304.     for (i = 0; i <= max; i++)
  305.     if (!loc[i])
  306.         loc[i] = cur;
  307.     Renew(loc,max+1,CMD*);    /* chop it down to size */
  308.     head->ucmd.scmd.sc_offset = min;
  309.     head->ucmd.scmd.sc_max = max;
  310.     head->ucmd.scmd.sc_next = loc;
  311. }
  312.  
  313. static void
  314. make_nswitch(head,count)
  315. register CMD *head;
  316. int count;
  317. {
  318.     register CMD *cur = head;
  319.     register CMD **loc;
  320.     register int i;
  321.     register int min = 32767;
  322.     register int max = -32768;
  323.     int origcount = count;
  324.     double value;        /* or your money back! */
  325.     short changed;        /* so triple your money back! */
  326.  
  327.     while (count--) {
  328.     i = (int)str_gnum(cur->c_short);
  329.     value = (double)i;
  330.     if (value != cur->c_short->str_u.str_nval)
  331.         return;        /* fractional values--just forget it */
  332.     changed = i;
  333.     if (changed != i)
  334.         return;        /* too big for a short */
  335.     if (cur->c_slen == O_LE)
  336.         i++;
  337.     else if (cur->c_slen == O_GE)    /* we only do < or > here */
  338.         i--;
  339.     if (i < min)
  340.         min = i;
  341.     if (i > max)
  342.         max = i;
  343.     cur = cur->c_next;
  344.     }
  345.     count = origcount;
  346.     if (max - min > count * 2 + 10)        /* too sparse? */
  347.     return;
  348.  
  349.     /* now make a new head in the exact same spot */
  350.     New(104,cur, 1, CMD);
  351.     StructCopy(head,cur,CMD);
  352.     Zero(head,1,CMD);
  353.     head->c_head = cur->c_head;
  354.     head->c_type = C_NSWITCH;
  355.     head->c_next = cur;        /* insert new cmd at front of list */
  356.     head->c_stab = cur->c_stab;
  357.  
  358.     Newz(105,loc, max - min + 3, CMD*);
  359.     loc++;
  360.     max -= min;
  361.     max++;
  362.     while (count--) {
  363.     i = (int)str_gnum(cur->c_short);
  364.     i -= min;
  365.     switch(cur->c_slen) {
  366.     case O_LE:
  367.         i++;
  368.     case O_LT:
  369.         for (i--; i >= -1; i--)
  370.         if (!loc[i])
  371.             loc[i] = cur;
  372.         break;
  373.     case O_GE:
  374.         i--;
  375.     case O_GT:
  376.         for (i++; i <= max; i++)
  377.         if (!loc[i])
  378.             loc[i] = cur;
  379.         break;
  380.     case O_EQ:
  381.         if (!loc[i])
  382.         loc[i] = cur;
  383.         break;
  384.     }
  385.     cur = cur->c_next;
  386.     }
  387.     loc--;
  388.     min--;
  389.     max++;
  390.     for (i = 0; i <= max; i++)
  391.     if (!loc[i])
  392.         loc[i] = cur;
  393.     head->ucmd.scmd.sc_offset = min;
  394.     head->ucmd.scmd.sc_max = max;
  395.     head->ucmd.scmd.sc_next = loc;
  396. }
  397.  
  398. CMD *
  399. append_line(head,tail)
  400. register CMD *head;
  401. register CMD *tail;
  402. {
  403.     if (tail == Nullcmd)
  404.     return head;
  405.     if (!tail->c_head)            /* make sure tail is well formed */
  406.     tail->c_head = tail;
  407.     if (head != Nullcmd) {
  408.     tail = tail->c_head;        /* get to start of tail list */
  409.     if (!head->c_head)
  410.         head->c_head = head;    /* start a new head list */
  411.     while (head->c_next) {
  412.         head->c_next->c_head = head->c_head;
  413.         head = head->c_next;    /* get to end of head list */
  414.     }
  415.     head->c_next = tail;        /* link to end of old list */
  416.     tail->c_head = head->c_head;    /* propagate head pointer */
  417.     }
  418.     while (tail->c_next) {
  419.     tail->c_next->c_head = tail->c_head;
  420.     tail = tail->c_next;
  421.     }
  422.     return tail;
  423. }
  424.  
  425. CMD *
  426. dodb(cur)
  427. CMD *cur;
  428. {
  429.     register CMD *cmd;
  430.     register CMD *head = cur->c_head;
  431.     STR *str;
  432.  
  433.     if (!head)
  434.     head = cur;
  435.     if (!head->c_line)
  436.     return cur;
  437.     str = afetch(stab_xarray(curcmd->c_filestab),(int)head->c_line,FALSE);
  438.     if (str == &str_undef || str->str_nok)
  439.     return cur;
  440.     str->str_u.str_nval = (double)head->c_line;
  441.     str->str_nok = 1;
  442.     Newz(106,cmd,1,CMD);
  443.     str_magic(str, curcmd->c_filestab, 0, Nullch, 0);
  444.     str->str_magic->str_u.str_cmd = cmd;
  445.     cmd->c_type = C_EXPR;
  446.     cmd->ucmd.acmd.ac_stab = Nullstab;
  447.     cmd->ucmd.acmd.ac_expr = Nullarg;
  448.     cmd->c_expr = make_op(O_SUBR, 2,
  449.     stab2arg(A_WORD,DBstab),
  450.     Nullarg,
  451.     Nullarg);
  452.     /*SUPPRESS 53*/
  453.     cmd->c_flags |= CF_COND|CF_DBSUB|CFT_D0;
  454.     cmd->c_line = head->c_line;
  455.     cmd->c_label = head->c_label;
  456.     cmd->c_filestab = curcmd->c_filestab;
  457.     cmd->c_stash = curstash;
  458.     return append_line(cmd, cur);
  459. }
  460.  
  461. CMD *
  462. make_acmd(type,stab,cond,arg)
  463. int type;
  464. STAB *stab;
  465. ARG *cond;
  466. ARG *arg;
  467. {
  468.     register CMD *cmd;
  469.  
  470.     Newz(107,cmd,1,CMD);
  471.     cmd->c_type = type;
  472.     cmd->ucmd.acmd.ac_stab = stab;
  473.     cmd->ucmd.acmd.ac_expr = arg;
  474.     cmd->c_expr = cond;
  475.     if (cond)
  476.     cmd->c_flags |= CF_COND;
  477.     if (cmdline == NOLINE)
  478.     cmd->c_line = curcmd->c_line;
  479.     else {
  480.     cmd->c_line = cmdline;
  481.     cmdline = NOLINE;
  482.     }
  483.     cmd->c_filestab = curcmd->c_filestab;
  484.     cmd->c_stash = curstash;
  485.     if (perldb)
  486.     cmd = dodb(cmd);
  487.     return cmd;
  488. }
  489.  
  490. CMD *
  491. make_ccmd(type,debuggable,arg,cblock)
  492. int type;
  493. int debuggable;
  494. ARG *arg;
  495. struct compcmd cblock;
  496. {
  497.     register CMD *cmd;
  498.  
  499.     Newz(108,cmd, 1, CMD);
  500.     cmd->c_type = type;
  501.     cmd->c_expr = arg;
  502.     cmd->ucmd.ccmd.cc_true = cblock.comp_true;
  503.     cmd->ucmd.ccmd.cc_alt = cblock.comp_alt;
  504.     if (arg)
  505.     cmd->c_flags |= CF_COND;
  506.     if (cmdline == NOLINE)
  507.     cmd->c_line = curcmd->c_line;
  508.     else {
  509.     cmd->c_line = cmdline;
  510.     cmdline = NOLINE;
  511.     }
  512.     cmd->c_filestab = curcmd->c_filestab;
  513.     cmd->c_stash = curstash;
  514.     if (perldb && debuggable)
  515.     cmd = dodb(cmd);
  516.     return cmd;
  517. }
  518.  
  519. CMD *
  520. make_icmd(type,arg,cblock)
  521. int type;
  522. ARG *arg;
  523. struct compcmd cblock;
  524. {
  525.     register CMD *cmd;
  526.     register CMD *alt;
  527.     register CMD *cur;
  528.     register CMD *head;
  529.     struct compcmd ncblock;
  530.  
  531.     Newz(109,cmd, 1, CMD);
  532.     head = cmd;
  533.     cmd->c_type = type;
  534.     cmd->c_expr = arg;
  535.     cmd->ucmd.ccmd.cc_true = cblock.comp_true;
  536.     cmd->ucmd.ccmd.cc_alt = cblock.comp_alt;
  537.     if (arg)
  538.     cmd->c_flags |= CF_COND;
  539.     if (cmdline == NOLINE)
  540.     cmd->c_line = curcmd->c_line;
  541.     else {
  542.     cmd->c_line = cmdline;
  543.     cmdline = NOLINE;
  544.     }
  545.     cmd->c_filestab = curcmd->c_filestab;
  546.     cmd->c_stash = curstash;
  547.     cur = cmd;
  548.     alt = cblock.comp_alt;
  549.     while (alt && alt->c_type == C_ELSIF) {
  550.     cur = alt;
  551.     alt = alt->ucmd.ccmd.cc_alt;
  552.     }
  553.     if (alt) {            /* a real life ELSE at the end? */
  554.     ncblock.comp_true = alt;
  555.     ncblock.comp_alt = Nullcmd;
  556.     alt = append_line(cur,make_ccmd(C_ELSE,1,Nullarg,ncblock));
  557.     cur->ucmd.ccmd.cc_alt = alt;
  558.     }
  559.     else
  560.     alt = cur;        /* no ELSE, so cur is proxy ELSE */
  561.  
  562.     cur = cmd;
  563.     while (cmd) {        /* now point everyone at the ELSE */
  564.     cur = cmd;
  565.     cmd = cur->ucmd.ccmd.cc_alt;
  566.     cur->c_head = head;
  567.     if (cur->c_type == C_ELSIF)
  568.         cur->c_type = C_IF;
  569.     if (cur->c_type == C_IF)
  570.         cur->ucmd.ccmd.cc_alt = alt;
  571.     if (cur == alt)
  572.         break;
  573.     cur->c_next = cmd;
  574.     }
  575.     if (perldb)
  576.     cur = dodb(cur);
  577.     return cur;
  578. }
  579.  
  580. void
  581. opt_arg(cmd,fliporflop,acmd)
  582. register CMD *cmd;
  583. int fliporflop;
  584. int acmd;
  585. {
  586.     register ARG *arg;
  587.     int opt = CFT_EVAL;
  588.     int sure = 0;
  589.     ARG *arg2;
  590.     int context = 0;    /* 0 = normal, 1 = before &&, 2 = before || */
  591.     int flp = fliporflop;
  592.  
  593.     if (!cmd)
  594.     return;
  595.     if (!(arg = cmd->c_expr)) {
  596.     cmd->c_flags &= ~CF_COND;
  597.     return;
  598.     }
  599.  
  600.     /* Can we turn && and || into if and unless? */
  601.  
  602.     if (acmd && !cmd->ucmd.acmd.ac_expr && !(cmd->c_flags & CF_TERM) &&
  603.       (arg->arg_type == O_AND || arg->arg_type == O_OR) ) {
  604.     dehoist(arg,1);
  605.     arg[2].arg_type &= A_MASK;    /* don't suppress eval */
  606.     dehoist(arg,2);
  607.     cmd->ucmd.acmd.ac_expr = arg[2].arg_ptr.arg_arg;
  608.     cmd->c_expr = arg[1].arg_ptr.arg_arg;
  609.     if (arg->arg_type == O_OR)
  610.         cmd->c_flags ^= CF_INVERT;        /* || is like unless */
  611.     arg->arg_len = 0;
  612.     free_arg(arg);
  613.     arg = cmd->c_expr;
  614.     }
  615.  
  616.     /* Turn "if (!expr)" into "unless (expr)" */
  617.  
  618.     if (!(cmd->c_flags & CF_TERM)) {        /* unless return value wanted */
  619.     while (arg->arg_type == O_NOT) {
  620.         dehoist(arg,1);
  621.         cmd->c_flags ^= CF_INVERT;        /* flip sense of cmd */
  622.         cmd->c_expr = arg[1].arg_ptr.arg_arg; /* hoist the rest of expr */
  623.         free_arg(arg);
  624.         arg = cmd->c_expr;            /* here we go again */
  625.     }
  626.     }
  627.  
  628.     if (!arg->arg_len) {        /* sanity check */
  629.     cmd->c_flags |= opt;
  630.     return;
  631.     }
  632.  
  633.     /* for "cond .. cond" we set up for the initial check */
  634.  
  635.     if (arg->arg_type == O_FLIP)
  636.     context |= 4;
  637.  
  638.     /* for "cond && expr" and "cond || expr" we can ignore expr, sort of */
  639.  
  640.   morecontext:
  641.     if (arg->arg_type == O_AND)
  642.     context |= 1;
  643.     else if (arg->arg_type == O_OR)
  644.     context |= 2;
  645.     if (context && (arg[flp].arg_type & A_MASK) == A_EXPR) {
  646.     arg = arg[flp].arg_ptr.arg_arg;
  647.     flp = 1;
  648.     if (arg->arg_type == O_AND || arg->arg_type == O_OR)
  649.         goto morecontext;
  650.     }
  651.     if ((context & 3) == 3)
  652.     return;
  653.  
  654.     if (arg[flp].arg_flags & (AF_PRE|AF_POST)) {
  655.     cmd->c_flags |= opt;
  656.     if (acmd && !cmd->ucmd.acmd.ac_expr && !(cmd->c_flags & CF_TERM)
  657.       && cmd->c_expr->arg_type == O_ITEM) {
  658.         arg[flp].arg_flags &= ~AF_POST;    /* prefer ++$foo to $foo++ */
  659.         arg[flp].arg_flags |= AF_PRE;    /*  if value not wanted */
  660.     }
  661.     return;                /* side effect, can't optimize */
  662.     }
  663.  
  664.     if (arg->arg_type == O_ITEM || arg->arg_type == O_FLIP ||
  665.       arg->arg_type == O_AND || arg->arg_type == O_OR) {
  666.     if ((arg[flp].arg_type & A_MASK) == A_SINGLE) {
  667.         opt = (str_true(arg[flp].arg_ptr.arg_str) ? CFT_TRUE : CFT_FALSE);
  668.         cmd->c_short = str_smake(arg[flp].arg_ptr.arg_str);
  669.         goto literal;
  670.     }
  671.     else if ((arg[flp].arg_type & A_MASK) == A_STAB ||
  672.       (arg[flp].arg_type & A_MASK) == A_LVAL) {
  673.         cmd->c_stab  = arg[flp].arg_ptr.arg_stab;
  674.         if (!context)
  675.         arg[flp].arg_ptr.arg_stab = Nullstab;
  676.         opt = CFT_REG;
  677.       literal:
  678.         if (!context) {    /* no && or ||? */
  679.         arg_free(arg);
  680.         cmd->c_expr = Nullarg;
  681.         }
  682.         if (!(context & 1))
  683.         cmd->c_flags |= CF_EQSURE;
  684.         if (!(context & 2))
  685.         cmd->c_flags |= CF_NESURE;
  686.     }
  687.     }
  688.     else if (arg->arg_type == O_MATCH || arg->arg_type == O_SUBST ||
  689.          arg->arg_type == O_NMATCH || arg->arg_type == O_NSUBST) {
  690.     if ((arg[1].arg_type == A_STAB || arg[1].arg_type == A_LVAL) &&
  691.         (arg[2].arg_type & A_MASK) == A_SPAT &&
  692.         arg[2].arg_ptr.arg_spat->spat_short &&
  693.         (arg->arg_type == O_SUBST || arg->arg_type == O_NSUBST ||
  694.          (arg[2].arg_ptr.arg_spat->spat_flags & SPAT_GLOBAL) == 0 )) {
  695.         cmd->c_stab  = arg[1].arg_ptr.arg_stab;
  696.         cmd->c_short = str_smake(arg[2].arg_ptr.arg_spat->spat_short);
  697.         cmd->c_slen  = arg[2].arg_ptr.arg_spat->spat_slen;
  698.         if (arg[2].arg_ptr.arg_spat->spat_flags & SPAT_ALL &&
  699.         !(arg[2].arg_ptr.arg_spat->spat_flags & SPAT_ONCE) &&
  700.         (arg->arg_type == O_MATCH || arg->arg_type == O_NMATCH) )
  701.         sure |= CF_EQSURE;        /* (SUBST must be forced even */
  702.                         /* if we know it will work.) */
  703.         if (arg->arg_type != O_SUBST) {
  704.         str_free(arg[2].arg_ptr.arg_spat->spat_short);
  705.         arg[2].arg_ptr.arg_spat->spat_short = Nullstr;
  706.         arg[2].arg_ptr.arg_spat->spat_slen = 0; /* only one chk */
  707.         }
  708.         sure |= CF_NESURE;        /* normally only sure if it fails */
  709.         if (arg->arg_type == O_NMATCH || arg->arg_type == O_NSUBST)
  710.         cmd->c_flags |= CF_FIRSTNEG;
  711.         if (context & 1) {        /* only sure if thing is false */
  712.         if (cmd->c_flags & CF_FIRSTNEG)
  713.             sure &= ~CF_NESURE;
  714.         else
  715.             sure &= ~CF_EQSURE;
  716.         }
  717.         else if (context & 2) {    /* only sure if thing is true */
  718.         if (cmd->c_flags & CF_FIRSTNEG)
  719.             sure &= ~CF_EQSURE;
  720.         else
  721.             sure &= ~CF_NESURE;
  722.         }
  723.         if (sure & (CF_EQSURE|CF_NESURE)) {    /* if we know anything*/
  724.         if (arg[2].arg_ptr.arg_spat->spat_flags & SPAT_SCANFIRST)
  725.             opt = CFT_SCAN;
  726.         else
  727.             opt = CFT_ANCHOR;
  728.         if (sure == (CF_EQSURE|CF_NESURE)    /* really sure? */
  729.             && arg->arg_type == O_MATCH
  730.             && context & 4
  731.             && fliporflop == 1) {
  732.             spat_free(arg[2].arg_ptr.arg_spat);
  733.             arg[2].arg_ptr.arg_spat = Nullspat;    /* don't do twice */
  734.         }
  735.         else
  736.             cmd->c_spat = arg[2].arg_ptr.arg_spat;
  737.         cmd->c_flags |= sure;
  738.         }
  739.     }
  740.     }
  741.     else if (arg->arg_type == O_SEQ || arg->arg_type == O_SNE ||
  742.          arg->arg_type == O_SLT || arg->arg_type == O_SGT) {
  743.     if (arg[1].arg_type == A_STAB || arg[1].arg_type == A_LVAL) {
  744.         if (arg[2].arg_type == A_SINGLE) {
  745.         /*SUPPRESS 594*/
  746.         char *junk = str_get(arg[2].arg_ptr.arg_str);
  747.  
  748.         cmd->c_stab  = arg[1].arg_ptr.arg_stab;
  749.         cmd->c_short = str_smake(arg[2].arg_ptr.arg_str);
  750.         cmd->c_slen  = cmd->c_short->str_cur+1;
  751.         switch (arg->arg_type) {
  752.         case O_SLT: case O_SGT:
  753.             sure |= CF_EQSURE;
  754.             cmd->c_flags |= CF_FIRSTNEG;
  755.             break;
  756.         case O_SNE:
  757.             cmd->c_flags |= CF_FIRSTNEG;
  758.             /* FALL THROUGH */
  759.         case O_SEQ:
  760.             sure |= CF_NESURE|CF_EQSURE;
  761.             break;
  762.         }
  763.         if (context & 1) {    /* only sure if thing is false */
  764.             if (cmd->c_flags & CF_FIRSTNEG)
  765.             sure &= ~CF_NESURE;
  766.             else
  767.             sure &= ~CF_EQSURE;
  768.         }
  769.         else if (context & 2) { /* only sure if thing is true */
  770.             if (cmd->c_flags & CF_FIRSTNEG)
  771.             sure &= ~CF_EQSURE;
  772.             else
  773.             sure &= ~CF_NESURE;
  774.         }
  775.         if (sure & (CF_EQSURE|CF_NESURE)) {
  776.             opt = CFT_STROP;
  777.             cmd->c_flags |= sure;
  778.         }
  779.         }
  780.     }
  781.     }
  782.     else if (arg->arg_type == O_EQ || arg->arg_type == O_NE ||
  783.          arg->arg_type == O_LE || arg->arg_type == O_GE ||
  784.          arg->arg_type == O_LT || arg->arg_type == O_GT) {
  785.     if (arg[1].arg_type == A_STAB || arg[1].arg_type == A_LVAL) {
  786.         if (arg[2].arg_type == A_SINGLE) {
  787.         cmd->c_stab  = arg[1].arg_ptr.arg_stab;
  788.         if (dowarn) {
  789.             STR *str = arg[2].arg_ptr.arg_str;
  790.  
  791.             if ((!str->str_nok && !looks_like_number(str)))
  792.             warn("Possible use of == on string value");
  793.         }
  794.         cmd->c_short = str_nmake(str_gnum(arg[2].arg_ptr.arg_str));
  795.         cmd->c_slen = arg->arg_type;
  796.         sure |= CF_NESURE|CF_EQSURE;
  797.         if (context & 1) {    /* only sure if thing is false */
  798.             sure &= ~CF_EQSURE;
  799.         }
  800.         else if (context & 2) { /* only sure if thing is true */
  801.             sure &= ~CF_NESURE;
  802.         }
  803.         if (sure & (CF_EQSURE|CF_NESURE)) {
  804.             opt = CFT_NUMOP;
  805.             cmd->c_flags |= sure;
  806.         }
  807.         }
  808.     }
  809.     }
  810.     else if (arg->arg_type == O_ASSIGN &&
  811.          (arg[1].arg_type == A_STAB || arg[1].arg_type == A_LVAL) &&
  812.          arg[1].arg_ptr.arg_stab == defstab &&
  813.          arg[2].arg_type == A_EXPR ) {
  814.     arg2 = arg[2].arg_ptr.arg_arg;
  815.     if (arg2->arg_type == O_ITEM && arg2[1].arg_type == A_READ) {
  816.         opt = CFT_GETS;
  817.         cmd->c_stab = arg2[1].arg_ptr.arg_stab;
  818.         if (!(stab_io(arg2[1].arg_ptr.arg_stab)->flags & IOF_ARGV)) {
  819.         free_arg(arg2);
  820.         arg[2].arg_ptr.arg_arg = Nullarg;
  821.         free_arg(arg);
  822.         cmd->c_expr = Nullarg;
  823.         }
  824.     }
  825.     }
  826.     else if (arg->arg_type == O_CHOP &&
  827.          (arg[1].arg_type == A_STAB || arg[1].arg_type == A_LVAL) ) {
  828.     opt = CFT_CHOP;
  829.     cmd->c_stab = arg[1].arg_ptr.arg_stab;
  830.     free_arg(arg);
  831.     cmd->c_expr = Nullarg;
  832.     }
  833.     if (context & 4)
  834.     opt |= CF_FLIP;
  835.     cmd->c_flags |= opt;
  836.  
  837.     if (cmd->c_flags & CF_FLIP) {
  838.     if (fliporflop == 1) {
  839.         arg = cmd->c_expr;    /* get back to O_FLIP arg */
  840.         New(110,arg[3].arg_ptr.arg_cmd, 1, CMD);
  841.         Copy(cmd, arg[3].arg_ptr.arg_cmd, 1, CMD);
  842.         New(111,arg[4].arg_ptr.arg_cmd,1,CMD);
  843.         Copy(cmd, arg[4].arg_ptr.arg_cmd, 1, CMD);
  844.         opt_arg(arg[4].arg_ptr.arg_cmd,2,acmd);
  845.         arg->arg_len = 2;        /* this is a lie */
  846.     }
  847.     else {
  848.         if ((opt & CF_OPTIMIZE) == CFT_EVAL)
  849.         cmd->c_flags = (cmd->c_flags & ~CF_OPTIMIZE) | CFT_UNFLIP;
  850.     }
  851.     }
  852. }
  853.  
  854. CMD *
  855. add_label(lbl,cmd)
  856. char *lbl;
  857. register CMD *cmd;
  858. {
  859.     if (cmd)
  860.     cmd->c_label = lbl;
  861.     return cmd;
  862. }
  863.  
  864. CMD *
  865. addcond(cmd, arg)
  866. register CMD *cmd;
  867. register ARG *arg;
  868. {
  869.     cmd->c_expr = arg;
  870.     cmd->c_flags |= CF_COND;
  871.     return cmd;
  872. }
  873.  
  874. CMD *
  875. addloop(cmd, arg)
  876. register CMD *cmd;
  877. register ARG *arg;
  878. {
  879.     void while_io();
  880.  
  881.     cmd->c_expr = arg;
  882.     cmd->c_flags |= CF_COND|CF_LOOP;
  883.  
  884.     if (!(cmd->c_flags & CF_INVERT))
  885.     while_io(cmd);        /* add $_ =, if necessary */
  886.  
  887.     if (cmd->c_type == C_BLOCK)
  888.     cmd->c_flags &= ~CF_COND;
  889.     else {
  890.     arg = cmd->ucmd.acmd.ac_expr;
  891.     if (arg && arg->arg_type == O_ITEM && arg[1].arg_type == A_CMD)
  892.         cmd->c_flags &= ~CF_COND;  /* "do {} while" happens at least once */
  893.     if (arg && (arg->arg_flags & AF_DEPR) &&
  894.       (arg->arg_type == O_SUBR || arg->arg_type == O_DBSUBR) )
  895.         cmd->c_flags &= ~CF_COND;  /* likewise for "do subr() while" */
  896.     }
  897.     return cmd;
  898. }
  899.  
  900. CMD *
  901. invert(cmd)
  902. CMD *cmd;
  903. {
  904.     register CMD *targ = cmd;
  905.     if (targ->c_head)
  906.     targ = targ->c_head;
  907.     if (targ->c_flags & CF_DBSUB)
  908.     targ = targ->c_next;
  909.     targ->c_flags ^= CF_INVERT;
  910.     return cmd;
  911. }
  912.  
  913. void
  914. cpy7bit(d,s,l)
  915. register char *d;
  916. register char *s;
  917. register int l;
  918. {
  919.     while (l--)
  920.     *d++ = *s++ & 127;
  921.     *d = '\0';
  922. }
  923.  
  924. int
  925. yyerror(s)
  926. char *s;
  927. {
  928.     char tmpbuf[258];
  929.     char tmp2buf[258];
  930.     char *tname = tmpbuf;
  931.  
  932.     if (bufptr > oldoldbufptr && bufptr - oldoldbufptr < 200 &&
  933.       oldoldbufptr != oldbufptr && oldbufptr != bufptr) {
  934.     while (isSPACE(*oldoldbufptr))
  935.         oldoldbufptr++;
  936.     cpy7bit(tmp2buf, oldoldbufptr, bufptr - oldoldbufptr);
  937.     sprintf(tname,"next 2 tokens \"%s\"",tmp2buf);
  938.     }
  939.     else if (bufptr > oldbufptr && bufptr - oldbufptr < 200 &&
  940.       oldbufptr != bufptr) {
  941.     while (isSPACE(*oldbufptr))
  942.         oldbufptr++;
  943.     cpy7bit(tmp2buf, oldbufptr, bufptr - oldbufptr);
  944.     sprintf(tname,"next token \"%s\"",tmp2buf);
  945.     }
  946.     else if (yychar > 256)
  947.     tname = "next token ???";
  948.     else if (!yychar)
  949.     (void)strcpy(tname,"at EOF");
  950.     else if (yychar < 32)
  951.     (void)sprintf(tname,"next char ^%c",yychar+64);
  952.     else if (yychar == 127)
  953.     (void)strcpy(tname,"at EOF");
  954.     else
  955.     (void)sprintf(tname,"next char %c",yychar);
  956. #ifdef macintosh
  957.     (void)sprintf(buf, "# %s, %s\n", s, tname);
  958.     if (curcmd->c_line == multi_end && multi_start < multi_end)
  959.     sprintf(buf+strlen(buf),
  960.       "#  (Might be a runaway multi-line %c%c string starting on line %d)\n",
  961.       multi_open,multi_close,multi_start);
  962.     strcpy(
  963.         MPWPosIndication(
  964.         buf+strlen(buf),
  965.         stab_val(curcmd->c_filestab)->str_ptr,
  966.         curcmd->c_line),
  967.     "\n");
  968. #else
  969.     (void)sprintf(buf, "%s in file %s at line %d, %s\n",
  970.       s,stab_val(curcmd->c_filestab)->str_ptr,curcmd->c_line,tname);
  971.     if (curcmd->c_line == multi_end && multi_start < multi_end)
  972.     sprintf(buf+strlen(buf),
  973.       "  (Might be a runaway multi-line %c%c string starting on line %d)\n",
  974.       multi_open,multi_close,multi_start);
  975. #endif
  976.     if (in_eval)
  977.     str_cat(stab_val(stabent("@",TRUE)),buf);
  978.     else
  979.     fputs(buf,stderr);
  980.     if (++error_count >= 10)
  981.     fatal("%s has too many errors.\n",
  982.     stab_val(curcmd->c_filestab)->str_ptr);
  983. }
  984.  
  985. void
  986. while_io(cmd)
  987. register CMD *cmd;
  988. {
  989.     register ARG *arg = cmd->c_expr;
  990.     STAB *asgnstab;
  991.  
  992.     /* hoist "while (<channel>)" up into command block */
  993.  
  994.     if (arg && arg->arg_type == O_ITEM && arg[1].arg_type == A_READ) {
  995.     cmd->c_flags &= ~CF_OPTIMIZE;    /* clear optimization type */
  996.     cmd->c_flags |= CFT_GETS;    /* and set it to do the input */
  997.     cmd->c_stab = arg[1].arg_ptr.arg_stab;
  998.     if (stab_io(arg[1].arg_ptr.arg_stab)->flags & IOF_ARGV) {
  999.         cmd->c_expr = l(make_op(O_ASSIGN, 2,    /* fake up "$_ =" */
  1000.            stab2arg(A_LVAL,defstab), arg, Nullarg));
  1001.     }
  1002.     else {
  1003.         free_arg(arg);
  1004.         cmd->c_expr = Nullarg;
  1005.     }
  1006.     }
  1007.     else if (arg && arg->arg_type == O_ITEM && arg[1].arg_type == A_INDREAD) {
  1008.     cmd->c_flags &= ~CF_OPTIMIZE;    /* clear optimization type */
  1009.     cmd->c_flags |= CFT_INDGETS;    /* and set it to do the input */
  1010.     cmd->c_stab = arg[1].arg_ptr.arg_stab;
  1011.     free_arg(arg);
  1012.     cmd->c_expr = Nullarg;
  1013.     }
  1014.     else if (arg && arg->arg_type == O_ITEM && arg[1].arg_type == A_GLOB) {
  1015.     if ((cmd->c_flags & CF_OPTIMIZE) == CFT_ARRAY)
  1016.         asgnstab = cmd->c_stab;
  1017.     else
  1018.         asgnstab = defstab;
  1019.     cmd->c_expr = l(make_op(O_ASSIGN, 2,    /* fake up "$foo =" */
  1020.        stab2arg(A_LVAL,asgnstab), arg, Nullarg));
  1021.     cmd->c_flags &= ~CF_OPTIMIZE;    /* clear optimization type */
  1022.     }
  1023. }
  1024.  
  1025. CMD *
  1026. wopt(cmd)
  1027. register CMD *cmd;
  1028. {
  1029.     register CMD *tail;
  1030.     CMD *newtail;
  1031.     register int i;
  1032.  
  1033.     if (cmd->c_expr && (cmd->c_flags & CF_OPTIMIZE) == CFT_FALSE)
  1034.     opt_arg(cmd,1, cmd->c_type == C_EXPR);
  1035.  
  1036.     while_io(cmd);        /* add $_ =, if necessary */
  1037.  
  1038.     /* First find the end of the true list */
  1039.  
  1040.     tail = cmd->ucmd.ccmd.cc_true;
  1041.     if (tail == Nullcmd)
  1042.     return cmd;
  1043.     New(112,newtail, 1, CMD);    /* guaranteed continue */
  1044.     for (;;) {
  1045.     /* optimize "next" to point directly to continue block */
  1046.     if (tail->c_type == C_EXPR &&
  1047.         tail->ucmd.acmd.ac_expr &&
  1048.         tail->ucmd.acmd.ac_expr->arg_type == O_NEXT &&
  1049.         (tail->ucmd.acmd.ac_expr->arg_len == 0 ||
  1050.          (cmd->c_label &&
  1051.           strEQ(cmd->c_label,
  1052.             tail->ucmd.acmd.ac_expr[1].arg_ptr.arg_str->str_ptr) )))
  1053.     {
  1054.         arg_free(tail->ucmd.acmd.ac_expr);
  1055.         tail->ucmd.acmd.ac_expr = Nullarg;
  1056.         tail->c_type = C_NEXT;
  1057.         if (cmd->ucmd.ccmd.cc_alt != Nullcmd)
  1058.         tail->ucmd.ccmd.cc_alt = cmd->ucmd.ccmd.cc_alt;
  1059.         else
  1060.         tail->ucmd.ccmd.cc_alt = newtail;
  1061.         tail->ucmd.ccmd.cc_true = Nullcmd;
  1062.     }
  1063.     else if (tail->c_type == C_IF && !tail->ucmd.ccmd.cc_alt) {
  1064.         if (cmd->ucmd.ccmd.cc_alt != Nullcmd)
  1065.         tail->ucmd.ccmd.cc_alt = cmd->ucmd.ccmd.cc_alt;
  1066.         else
  1067.         tail->ucmd.ccmd.cc_alt = newtail;
  1068.     }
  1069.     else if (tail->c_type == C_CSWITCH || tail->c_type == C_NSWITCH) {
  1070.         if (cmd->ucmd.ccmd.cc_alt != Nullcmd) {
  1071.         for (i = tail->ucmd.scmd.sc_max; i >= 0; i--)
  1072.             if (!tail->ucmd.scmd.sc_next[i])
  1073.             tail->ucmd.scmd.sc_next[i] = cmd->ucmd.ccmd.cc_alt;
  1074.         }
  1075.         else {
  1076.         for (i = tail->ucmd.scmd.sc_max; i >= 0; i--)
  1077.             if (!tail->ucmd.scmd.sc_next[i])
  1078.             tail->ucmd.scmd.sc_next[i] = newtail;
  1079.         }
  1080.     }
  1081.  
  1082.     if (!tail->c_next)
  1083.         break;
  1084.     tail = tail->c_next;
  1085.     }
  1086.  
  1087.     /* if there's a continue block, link it to true block and find end */
  1088.  
  1089.     if (cmd->ucmd.ccmd.cc_alt != Nullcmd) {
  1090. #ifdef macintosh
  1091.     /* Work around a code generation bug in MPW C 3.2 */
  1092.     tail->c_next = cmd->ucmd.ccmd.cc_alt;
  1093.     tail = cmd->ucmd.ccmd.cc_alt;
  1094. #else
  1095.     tail->c_next = cmd->ucmd.ccmd.cc_alt;
  1096.     tail = tail->c_next;
  1097. #endif
  1098.     for (;;) {
  1099.         /* optimize "next" to point directly to continue block */
  1100.         if (tail->c_type == C_EXPR &&
  1101.         tail->ucmd.acmd.ac_expr &&
  1102.         tail->ucmd.acmd.ac_expr->arg_type == O_NEXT &&
  1103.         (tail->ucmd.acmd.ac_expr->arg_len == 0 ||
  1104.          (cmd->c_label &&
  1105.           strEQ(cmd->c_label,
  1106.             tail->ucmd.acmd.ac_expr[1].arg_ptr.arg_str->str_ptr) )))
  1107.         {
  1108.         arg_free(tail->ucmd.acmd.ac_expr);
  1109.         tail->ucmd.acmd.ac_expr = Nullarg;
  1110.         tail->c_type = C_NEXT;
  1111.         tail->ucmd.ccmd.cc_alt = newtail;
  1112.         tail->ucmd.ccmd.cc_true = Nullcmd;
  1113.         }
  1114.         else if (tail->c_type == C_IF && !tail->ucmd.ccmd.cc_alt) {
  1115.         tail->ucmd.ccmd.cc_alt = newtail;
  1116.         }
  1117.         else if (tail->c_type == C_CSWITCH || tail->c_type == C_NSWITCH) {
  1118.         for (i = tail->ucmd.scmd.sc_max; i >= 0; i--)
  1119.             if (!tail->ucmd.scmd.sc_next[i])
  1120.             tail->ucmd.scmd.sc_next[i] = newtail;
  1121.         }
  1122.  
  1123.         if (!tail->c_next)
  1124.         break;
  1125.         tail = tail->c_next;
  1126.     }
  1127.     /*SUPPRESS 530*/
  1128.     for ( ; tail->c_next; tail = tail->c_next) ;
  1129.     }
  1130.  
  1131.     /* Here's the real trick: link the end of the list back to the beginning,
  1132.      * inserting a "last" block to break out of the loop.  This saves one or
  1133.      * two procedure calls every time through the loop, because of how cmd_exec
  1134.      * does tail recursion.
  1135.      */
  1136.  
  1137.     tail->c_next = newtail;
  1138.     tail = newtail;
  1139.     if (!cmd->ucmd.ccmd.cc_alt)
  1140.     cmd->ucmd.ccmd.cc_alt = tail;    /* every loop has a continue now */
  1141.  
  1142. #ifndef lint
  1143.     Copy((char *)cmd, (char *)tail, 1, CMD);
  1144. #endif
  1145.     tail->c_type = C_EXPR;
  1146.     tail->c_flags ^= CF_INVERT;        /* turn into "last unless" */
  1147.     tail->c_next = tail->ucmd.ccmd.cc_true;    /* loop directly back to top */
  1148.     tail->ucmd.acmd.ac_expr = make_op(O_LAST,0,Nullarg,Nullarg,Nullarg);
  1149.     tail->ucmd.acmd.ac_stab = Nullstab;
  1150.     return cmd;
  1151. }
  1152.  
  1153. CMD *
  1154. over(eachstab,cmd)
  1155. STAB *eachstab;
  1156. register CMD *cmd;
  1157. {
  1158.     /* hoist "for $foo (@bar)" up into command block */
  1159.  
  1160.     cmd->c_flags &= ~CF_OPTIMIZE;    /* clear optimization type */
  1161.     cmd->c_flags |= CFT_ARRAY;        /* and set it to do the iteration */
  1162.     cmd->c_stab = eachstab;
  1163.     cmd->c_short = Str_new(23,0);    /* just to save a field in struct cmd */
  1164.     cmd->c_short->str_u.str_useful = -1;
  1165.  
  1166.     return cmd;
  1167. }
  1168.  
  1169. void
  1170. cmd_free(cmd)
  1171. register CMD *cmd;
  1172. {
  1173.     register CMD *tofree;
  1174.     register CMD *head = cmd;
  1175.  
  1176.     if (!cmd)
  1177.     return;
  1178.     if (cmd->c_head != cmd)
  1179.     warn("Malformed cmd links\n");
  1180.     while (cmd) {
  1181.     if (cmd->c_type != C_WHILE) {    /* WHILE block is duplicated */
  1182.         if (cmd->c_label) {
  1183.         Safefree(cmd->c_label);
  1184.         cmd->c_label = Nullch;
  1185.         }
  1186.         if (cmd->c_short) {
  1187.         str_free(cmd->c_short);
  1188.         cmd->c_short = Nullstr;
  1189.         }
  1190.         if (cmd->c_expr) {
  1191.         arg_free(cmd->c_expr);
  1192.         cmd->c_expr = Nullarg;
  1193.         }
  1194.     }
  1195.     switch (cmd->c_type) {
  1196.     case C_WHILE:
  1197.     case C_BLOCK:
  1198.     case C_ELSE:
  1199.     case C_IF:
  1200.         if (cmd->ucmd.ccmd.cc_true) {
  1201.         cmd_free(cmd->ucmd.ccmd.cc_true);
  1202.         cmd->ucmd.ccmd.cc_true = Nullcmd;
  1203.         }
  1204.         break;
  1205.     case C_EXPR:
  1206.         if (cmd->ucmd.acmd.ac_expr) {
  1207.         arg_free(cmd->ucmd.acmd.ac_expr);
  1208.         cmd->ucmd.acmd.ac_expr = Nullarg;
  1209.         }
  1210.         break;
  1211.     }
  1212.     tofree = cmd;
  1213.     cmd = cmd->c_next;
  1214.     if (tofree != head)        /* to get Saber to shut up */
  1215.         Safefree(tofree);
  1216.     if (cmd && cmd == head)        /* reached end of while loop */
  1217.         break;
  1218.     }
  1219.     Safefree(head);
  1220. }
  1221.  
  1222. void
  1223. arg_free(arg)
  1224. register ARG *arg;
  1225. {
  1226.     register int i;
  1227.  
  1228.     if (!arg)
  1229.     return;
  1230.     for (i = 1; i <= arg->arg_len; i++) {
  1231.     switch (arg[i].arg_type & A_MASK) {
  1232.     case A_NULL:
  1233.         if (arg->arg_type == O_TRANS) {
  1234.         Safefree(arg[i].arg_ptr.arg_cval);
  1235.         arg[i].arg_ptr.arg_cval = Nullch;
  1236.         }
  1237.         break;
  1238.     case A_LEXPR:
  1239.         if (arg->arg_type == O_AASSIGN &&
  1240.           arg[i].arg_ptr.arg_arg->arg_type == O_LARRAY) {
  1241.         char *name = 
  1242.           stab_name(arg[i].arg_ptr.arg_arg[1].arg_ptr.arg_stab);
  1243.  
  1244.         if (strnEQ("_GEN_",name, 5))    /* array for foreach */
  1245.             hdelete(defstash,name,strlen(name));
  1246.         }
  1247.         /* FALL THROUGH */
  1248.     case A_EXPR:
  1249.         arg_free(arg[i].arg_ptr.arg_arg);
  1250.         arg[i].arg_ptr.arg_arg = Nullarg;
  1251.         break;
  1252.     case A_CMD:
  1253.         cmd_free(arg[i].arg_ptr.arg_cmd);
  1254.         arg[i].arg_ptr.arg_cmd = Nullcmd;
  1255.         break;
  1256.     case A_WORD:
  1257.     case A_STAB:
  1258.     case A_LVAL:
  1259.     case A_READ:
  1260.     case A_GLOB:
  1261.     case A_ARYLEN:
  1262.     case A_LARYLEN:
  1263.     case A_ARYSTAB:
  1264.     case A_LARYSTAB:
  1265.         break;
  1266.     case A_SINGLE:
  1267.     case A_DOUBLE:
  1268.     case A_BACKTICK:
  1269.         str_free(arg[i].arg_ptr.arg_str);
  1270.         arg[i].arg_ptr.arg_str = Nullstr;
  1271.         break;
  1272.     case A_SPAT:
  1273.         spat_free(arg[i].arg_ptr.arg_spat);
  1274.         arg[i].arg_ptr.arg_spat = Nullspat;
  1275.         break;
  1276.     }
  1277.     }
  1278.     free_arg(arg);
  1279. }
  1280.  
  1281. void
  1282. spat_free(spat)
  1283. register SPAT *spat;
  1284. {
  1285.     register SPAT *sp;
  1286.     HENT *entry;
  1287.  
  1288.     if (!spat)
  1289.     return;
  1290.     if (spat->spat_runtime) {
  1291.     arg_free(spat->spat_runtime);
  1292.     spat->spat_runtime = Nullarg;
  1293.     }
  1294.     if (spat->spat_repl) {
  1295.     arg_free(spat->spat_repl);
  1296.     spat->spat_repl = Nullarg;
  1297.     }
  1298.     if (spat->spat_short) {
  1299.     str_free(spat->spat_short);
  1300.     spat->spat_short = Nullstr;
  1301.     }
  1302.     if (spat->spat_regexp) {
  1303.     regfree(spat->spat_regexp);
  1304.     spat->spat_regexp = Null(REGEXP*);
  1305.     }
  1306.  
  1307.     /* now unlink from spat list */
  1308.  
  1309.     for (entry = defstash->tbl_array['_']; entry; entry = entry->hent_next) {
  1310.     register HASH *stash;
  1311.     STAB *stab = (STAB*)entry->hent_val;
  1312.  
  1313.     if (!stab)
  1314.         continue;
  1315.     stash = stab_hash(stab);
  1316.     if (!stash || stash->tbl_spatroot == Null(SPAT*))
  1317.         continue;
  1318.     if (stash->tbl_spatroot == spat)
  1319.         stash->tbl_spatroot = spat->spat_next;
  1320.     else {
  1321.         for (sp = stash->tbl_spatroot;
  1322.           sp && sp->spat_next != spat;
  1323.           sp = sp->spat_next)
  1324.         /*SUPPRESS 530*/
  1325.         ;
  1326.         if (sp)
  1327.         sp->spat_next = spat->spat_next;
  1328.     }
  1329.     }
  1330.     Safefree(spat);
  1331. }
  1332.  
  1333. /* Recursively descend a command sequence and push the address of any string
  1334.  * that needs saving on recursion onto the tosave array.
  1335.  */
  1336.  
  1337. static int
  1338. cmd_tosave(cmd,willsave)
  1339. register CMD *cmd;
  1340. int willsave;                /* willsave passes down the tree */
  1341. {
  1342.     register CMD *head = cmd;
  1343.     int shouldsave = FALSE;        /* shouldsave passes up the tree */
  1344.     int tmpsave;
  1345.     register CMD *lastcmd = Nullcmd;
  1346.  
  1347.     while (cmd) {
  1348.     if (cmd->c_expr)
  1349.         shouldsave |= arg_tosave(cmd->c_expr,willsave);
  1350.     switch (cmd->c_type) {
  1351.     case C_WHILE:
  1352.         if (cmd->ucmd.ccmd.cc_true) {
  1353.         tmpsave = cmd_tosave(cmd->ucmd.ccmd.cc_true,willsave);
  1354.  
  1355.         /* Here we check to see if the temporary array generated for
  1356.          * a foreach needs to be localized because of recursion.
  1357.          */
  1358.         if (tmpsave && (cmd->c_flags & CF_OPTIMIZE) == CFT_ARRAY) {
  1359.             if (lastcmd &&
  1360.               lastcmd->c_type == C_EXPR &&
  1361.               lastcmd->c_expr) {
  1362.             ARG *arg = lastcmd->c_expr;
  1363.  
  1364.             if (arg->arg_type == O_ASSIGN &&
  1365.                 arg[1].arg_type == A_LEXPR &&
  1366.                 arg[1].arg_ptr.arg_arg->arg_type == O_LARRAY &&
  1367.                 strnEQ("_GEN_",
  1368.                   stab_name(
  1369.                 arg[1].arg_ptr.arg_arg[1].arg_ptr.arg_stab),
  1370.                   5)) {    /* array generated for foreach */
  1371.                 (void)localize(arg);
  1372.             }
  1373.             }
  1374.  
  1375.             /* in any event, save the iterator */
  1376.  
  1377.             if (cmd->c_short)  /* Better safe than sorry */
  1378.             (void)apush(tosave,cmd->c_short);
  1379.         }
  1380.         shouldsave |= tmpsave;
  1381.         }
  1382.         break;
  1383.     case C_BLOCK:
  1384.     case C_ELSE:
  1385.     case C_IF:
  1386.         if (cmd->ucmd.ccmd.cc_true)
  1387.         shouldsave |= cmd_tosave(cmd->ucmd.ccmd.cc_true,willsave);
  1388.         break;
  1389.     case C_EXPR:
  1390.         if (cmd->ucmd.acmd.ac_expr)
  1391.         shouldsave |= arg_tosave(cmd->ucmd.acmd.ac_expr,willsave);
  1392.         break;
  1393.     }
  1394.     lastcmd = cmd;
  1395.     cmd = cmd->c_next;
  1396.     if (cmd && cmd == head)        /* reached end of while loop */
  1397.         break;
  1398.     }
  1399.     return shouldsave;
  1400. }
  1401.  
  1402. static int
  1403. arg_tosave(arg,willsave)
  1404. register ARG *arg;
  1405. int willsave;
  1406. {
  1407.     register int i;
  1408.     int shouldsave = FALSE;
  1409.  
  1410.     for (i = arg->arg_len; i >= 1; i--) {
  1411.     switch (arg[i].arg_type & A_MASK) {
  1412.     case A_NULL:
  1413.         break;
  1414.     case A_LEXPR:
  1415.     case A_EXPR:
  1416.         shouldsave |= arg_tosave(arg[i].arg_ptr.arg_arg,shouldsave);
  1417.         break;
  1418.     case A_CMD:
  1419.         shouldsave |= cmd_tosave(arg[i].arg_ptr.arg_cmd,shouldsave);
  1420.         break;
  1421.     case A_WORD:
  1422.     case A_STAB:
  1423.     case A_LVAL:
  1424.     case A_READ:
  1425.     case A_GLOB:
  1426.     case A_ARYLEN:
  1427.     case A_SINGLE:
  1428.     case A_DOUBLE:
  1429.     case A_BACKTICK:
  1430.         break;
  1431.     case A_SPAT:
  1432.         shouldsave |= spat_tosave(arg[i].arg_ptr.arg_spat);
  1433.         break;
  1434.     }
  1435.     }
  1436.     switch (arg->arg_type) {
  1437.     case O_RETURN:
  1438.     saw_return = TRUE;
  1439.     break;
  1440.     case O_EVAL:
  1441.     case O_SUBR:
  1442.     shouldsave = TRUE;
  1443.     break;
  1444.     }
  1445.     if (willsave && arg->arg_ptr.arg_str)
  1446.     (void)apush(tosave,arg->arg_ptr.arg_str);
  1447.     return shouldsave;
  1448. }
  1449.  
  1450. static int
  1451. spat_tosave(spat)
  1452. register SPAT *spat;
  1453. {
  1454.     int shouldsave = FALSE;
  1455.  
  1456.     if (spat->spat_runtime)
  1457.     shouldsave |= arg_tosave(spat->spat_runtime,FALSE);
  1458.     if (spat->spat_repl) {
  1459.     shouldsave |= arg_tosave(spat->spat_repl,FALSE);
  1460.     }
  1461.  
  1462.     return shouldsave;
  1463. }
  1464.  
  1465.